from ultralytics import YOLO
a1 = YOLO('yolo11n.pt')
a1.train(data="4.yaml",  # 数据集配置文件路径
         epochs=20,  # 训练轮次
         imgsz=640,   # 输出图片尺寸，官推640
         batch=20,   # 每次训练的批量
         device='cpu',   # 使用cpu还是使用gpu
         )
print("模型训练完毕")

